package leetcode.tencent;

import java.util.ArrayList;
import java.util.List;

/**
 * @author Cheng Jun
 * Description: 给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。
 * https://leetcode-cn.com/problems/spiral-matrix/
 * <p>
 * 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
 * 输出：[1,2,3,6,9,8,7,4,5]
 * @version 1.0
 * @date 2021/12/7 15:51
 * 二刷 面试题
 */
public class spiralOrder {

    public static void main(String[] args) {
        spiralOrder1(new int[][]{
                new int[]{1, 2, 3, 4},
                new int[]{5, 6, 7, 8},
                new int[]{9, 10, 11, 12}
        });
    }

    // 解题思路：转圈遍历，矩阵四条边为一圈，一圈接一圈地遍历
    static List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> res = new ArrayList<>();
        boolean[][] used = new boolean[matrix.length][matrix[0].length];
        // 行列长度
        int row = matrix.length;
        int column = matrix[0].length;

        int rowIndex = 0;
        int colIndex = 0;
        int times = 1;
        while (rowIndex < row && colIndex < column && !used[rowIndex][colIndex]) {
            // 从左往右
            for (int j = colIndex; j < column - times; j++) {
                if (!used[rowIndex][j]) {
                    res.add(matrix[rowIndex][j]);
                    used[rowIndex][j] = true;
                    colIndex = j + 1;
                }
            }
            // 从上往下
            for (int i = rowIndex; i < row - times; i++) {
                if (!used[i][colIndex]) {
                    res.add(matrix[i][colIndex]);
                    used[i][colIndex] = true;
                    rowIndex = i + 1;
                }
            }
            // 从右往左
            for (int j = colIndex; times - 1 < j; j--) {
                if (!used[rowIndex][j]) {
                    res.add(matrix[rowIndex][j]);
                    used[rowIndex][j] = true;
                    colIndex = j - 1;
                }
            }
            // 从下往上
            for (int i = rowIndex; times - 1 < i; i--) {
                if (!used[i][colIndex]) {
                    res.add(matrix[i][colIndex]);
                    used[i][colIndex] = true;
                    rowIndex = i - 1;
                }
            }
            rowIndex++;
            colIndex++;
            times++;
        }
        // 奇数情况 特殊处理
        if (row == column && (row & 1) == 1) {
            res.add(matrix[row / 2][column / 2]);
        }
        return res;
    }

    static List<Integer> spiralOrder1(int[][] matrix) {
        List<Integer> res = new ArrayList<>();
        // 行列长度
        int m = matrix.length;
        int n = matrix[0].length;
        int top = 0;
        int bottom = m - 1;
        int left = 0;
        int right = n - 1;
        // 计数器
        int counter = 0;
        while (counter < m * n) {
            // 注意每个循环里面 都要保证 counter < m * n，这样保证当循环完成后，后面的for 不可能再执行
            for (int i = left; i <= right && counter < m * n; i++, counter++) res.add(matrix[top][i]);
            top++;
            for (int i = top; i <= bottom && counter < m * n; i++, counter++) res.add(matrix[i][right]);
            right--;
            for (int i = right; i >= left && counter < m * n; i--, counter++) res.add(matrix[bottom][i]);
            bottom--;
            for (int i = bottom; i >= top && counter < m * n; i--, counter++) res.add(matrix[i][left]);
            left++;
        }
        return res;
    }

}
